package com.ht.web360.PojServer.dao;

import cn.hutool.core.util.NumberUtil;
import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import com.ht.web360.base.dto.PojInfo.dto.ObjRefDto;
import lombok.Data;

import java.time.LocalDateTime;

/**
 * 项目单元对对象的引用
 */
@Data
@TableName("obj_ref")
public class ObjRef {
    @TableId
    Long id;

    Long unitId;

    Long objId;

    @TableField(fill = FieldFill.INSERT)
    LocalDateTime createTime;

    public static ObjRef fromDto(ObjRefDto dto) {
        ObjRef objRef = new ObjRef();
        if (NumberUtil.isLong(dto.getId())) {
            objRef.id = Long.valueOf(dto.getId());
        }
        if (NumberUtil.isLong(dto.getPojId())) {
            objRef.unitId = Long.valueOf(dto.getPojId());
        }
        if (NumberUtil.isLong(dto.getObjId())) {
            objRef.objId = Long.valueOf(dto.getObjId());
        }
        return objRef;
    }

    public ObjRefDto toDto() {
        ObjRefDto dto = new ObjRefDto();
        dto.setId(this.id.toString());
        dto.setPojId(this.unitId.toString());
        dto.setObjId(this.objId.toString());
        return dto;
    }
}
